How to create 3D array in numpy Python